import numpy as np
import matplotlib.pyplot as plt
from opticspy import lens

def create_simple_lens_system():
    # 创建一个简单的光学系统
    system = lens.Lens()
    
    # 添加表面
    system.add_surface(radius=50, thickness=5, material='BK7')
    system.add_surface(radius=-50, thickness=100)
    
    return system

def main():
    # 创建光学系统
    system = create_simple_lens_system()
    
    # 打印系统信息
    print("光学系统信息：")
    print(f"表面数量: {len(system.surfaces)}")
    print(f"总厚度: {system.total_thickness} mm")
    
    # 绘制系统示意图
    system.plot()
    plt.title("简单透镜系统")
    plt.show()

if __name__ == "__main__":
    main() 